package com.shm.leetcode;

/**
 * 785. 判断二分图
 * 存在一个 无向图 ，图中有 n 个节点。其中每个节点都有一个介于 0 到 n - 1 之间的唯一编号。给你一个二维数组 graph ，其中 graph[u] 是一个节点数组，由节点 u 的邻接节点组成。形式上，对于 graph[u] 中的每个 v ，都存在一条位于节点 u 和节点 v 之间的无向边。该无向图同时具有以下属性：
 * 不存在自环（graph[u] 不包含 u）。
 * 不存在平行边（graph[u] 不包含重复值）。
 * 如果 v 在 graph[u] 内，那么 u 也应该在 graph[v] 内（该图是无向图）
 * 这个图可能不是连通图，也就是说两个节点 u 和 v 之间可能不存在一条连通彼此的路径。
 * 二分图 定义：如果能将一个图的节点集合分割成两个独立的子集 A 和 B ，并使图中的每一条边的两个节点一个来自 A 集合，一个来自 B 集合，就将这个图称为 二分图 。
 *
 * 如果图是二分图，返回 true ；否则，返回 false 。
 *
 *
 *
 * 示例 1：
 *
 *
 * 输入：graph = [[1,2,3],[0,2],[0,1,3],[0,2]]
 * 输出：false
 * 解释：不能将节点分割成两个独立的子集，以使每条边都连通一个子集中的一个节点与另一个子集中的一个节点。
 * 示例 2：
 *
 *
 * 输入：graph = [[1,3],[0,2],[1,3],[0,2]]
 * 输出：true
 * 解释：可以将节点分成两组: {0, 2} 和 {1, 3} 。
 *
 *
 * 提示：
 *
 * graph.length == n
 * 1 <= n <= 100
 * 0 <= graph[u].length < n
 * 0 <= graph[u][i] <= n - 1
 * graph[u] 不会包含 u
 * graph[u] 的所有值 互不相同
 * 如果 graph[u] 包含 v，那么 graph[v] 也会包含 u
 * @author SHM
 */
public class IsBipartite {
    /**
     * Java
     * 方法一：并查集
     * 遍历每一个顶点的邻接点集合，将其在并查集中合并
     * 并且在遍历过程中检查，如果当前顶点的某一邻接点已和当前顶点连通，那么就说明不是二分图
     * @param graph
     * @return
     */
    public boolean isBipartite(int[][] graph) {
        int n = graph.length;
        UnionFind uf = new UnionFind(n);
        for(int i=0;i<n;i++){
            //将一个顶点对应的所有邻接点合并
            int[] ver = graph[i];
            for(int v:ver){
                //如果邻接点和当前顶点连通，则不符合二分图定义
                if(uf.connect(v,i)){
                    return false;
                }
                uf.union(v,ver[0]);
            }
        }

        return true;
    }

    class UnionFind{
        int[] parent;
        int count;

        UnionFind(int n){
            parent = new int[n];
            count = n;
            for(int i=0;i<n;i++){
                parent[i] = i;
            }
        }

        int find(int x){
            if(x!=parent[x]){
                parent[x] = find(parent[x]);
            }
            return parent[x];
        }

        void union(int x,int y){
            int newX = find(x);
            int newY = find(y);
            if(newX==newY){
                return;
            }
            parent[newY] = newX;
            count--;
        }

        boolean connect(int x,int y){
            int newX = find(x);
            int newY = find(y);
            return newX == newY;
        }
    }
}
